package com.ld.producer.interceptor;

import org.apache.kafka.clients.producer.ProducerInterceptor;
import org.apache.kafka.clients.producer.ProducerRecord;
import org.apache.kafka.clients.producer.RecordMetadata;

import java.util.Map;

/**
 * @author ld
 *
 * 自定义一个生产者拦截器
 */
public class MyProducerInterceptor implements ProducerInterceptor<String, String> {

    /**
     * 此方法用于发送消息之前调用
     * @param producerRecord 消息对象
     * @return
     */
    @Override
    public ProducerRecord<String, String> onSend(ProducerRecord<String, String> producerRecord) {
        System.out.println("准备发送消息......");
        /**
         * 这里可以修改要发送的消息，例如为消息添加前缀 test-
         */
        //获取消息
        String value = producerRecord.value();
        //添加前缀
        value = "test-" + value;
        return new ProducerRecord<String,String>(producerRecord.topic(),value);
    }

    /**
     * 此方法用于发送消息之后,发送结果被应答之前调用
     * @param recordMetadata 发送消息成功，返回的发送结果
     * @param e 发送失败，返回的异常信息
     */
    @Override
    public void onAcknowledgement(RecordMetadata recordMetadata, Exception e) {
        /**
         * 可以判断发送成功还是发送失败
         */
        if(recordMetadata != null){
            System.out.println("发送消息成功... topic: " + recordMetadata.topic() + " , partition: " + recordMetadata.partition());
        }else{
            System.out.println("发送消息失败... 失败原因：" + e.getMessage());
        }
    }

    /**
     * 此方法用于在关闭拦截器时调用
     */
    @Override
    public void close() {
        System.out.println("关闭拦截器......");
    }

    @Override
    public void configure(Map<String, ?> map) {

    }
}
